"""


Version: 0.1
Author: lk
Date: 2022-03-08 19:21
"""
import numpy as np
import torch
import torch.nn as nn
import dgl
import torch.nn.functional as F
from dgl.nn.pytorch.conv import GINConv
from dgl.nn.pytorch.glob import AvgPooling, MaxPooling, SumPooling

if __name__ == '__main__':
    g=dgl.rand_graph(10,13)
    layer = GINConv(
        nn.Sigmoid(),
        "sum",
        0,
        False
    )
    while True:
        g_node_feats = torch.randn(10, 3)
        print(layer(g,g_node_feats))
